# from gym_pih.envs.wrapped_env import make_wrapped_env
# from gym_pih.envs.typec import TypecEnv
# from gym_pih.envs.typec_v2 import TypecEnv as TypecEnv_v2
# from gym_pih.envs.typec_v3 import TypecEnv as TypecEnv_v3
# from gym_pih.envs.typec_v4 import TypecEnv as TypecEnv_v4
# from gym_pih.envs.typec_v5 import TypecEnv as TypecEnv_v5
# from gym_pih.envs.two_stage import TwoStage as TwoStageEnv

def _lazy_import(name):
    import importlib
    module_name, class_name = name.rsplit('.', 1)
    module = importlib.import_module(module_name)
    return getattr(module, class_name)

# 供外部调用
make_wrapped_env = _lazy_import('gym_pih.envs.wrapped_env.make_wrapped_env')
TypecEnv = _lazy_import('gym_pih.envs.typec.TypecEnv')
TypecEnv_v2 = _lazy_import('gym_pih.envs.typec_v2.TypecEnv')
TypecEnv_v3 = _lazy_import('gym_pih.envs.typec_v3.TypecEnv')
TypecEnv_v4 = _lazy_import('gym_pih.envs.typec_v4.TypecEnv')
TypecEnv_v5 = _lazy_import('gym_pih.envs.typec_v5.TypecEnv')
TwoStageEnv = _lazy_import('gym_pih.envs.two_stage.TwoStage')